package leetcode;

import java.util.LinkedList;
import java.util.Queue;

/**
 * 958.二叉树的完全性检验
 * 给定一个二叉树，判断它是否为完全二叉树。
 */
public class Num_958 {
    /**
     * 层序遍历二叉树的每个有左右子树的节点，当遇到
     *    a.第一个只有左数没有右数的节点
     *    b.第一个叶子节点
     * 时，之后的节点必须都是叶子节点。
     * @param root
     * @return
     */
    public boolean isCompleteTree(TreeNode root) {
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        boolean isSecondStep = false;
        while(!queue.isEmpty()){
            TreeNode cur = queue.poll();
            /** 考察当前节点 **/
            //此时如果是第二状态，之后的节点必须都是叶子节点
            if(isSecondStep){
                if(cur.left != null || cur.right != null){
                    return false;
                }
            }else{
                //不是第二状态
                if(cur.left == null && cur.right != null){
                    //没有左树，但有右树，，一定不是完全二叉树
                    return false;
                }else if(cur.left != null && cur.right != null){
                    //既有左树，又有右树
                    queue.offer(cur.left);
                    queue.offer(cur.right);
                }else if(cur.left != null && cur.right == null){
                    //有左树，但没有右树
                    queue.offer(cur.left);
                    isSecondStep = true;
                }else if(cur.left == null && cur.right == null){
                    //该节点是叶子节点
                    isSecondStep = true;
                }
            }
        }
        return true;
    }
}
